import logging  # python库自带的
import logging.handlers
from flask import request
import os


class RequestShoppingFormatter(logging.Formatter):
    '''
    自定义日志输出格式，即要输出的内容
    '''

    def format(self, record):
        record.url = request.url  # 需要在日志中记录请求地址
        record.remote_addr = request.remote_addr  # 需要在日志中记录客户端的地址
        return super().format(record)  # 仍然使用父类中的format函数，重写format函数只是给record中添加了新的值


# 创建一个个性化的logger对象
def create_logger(app):
    '''
    设置日志配置
    :param app: Flask中的app对象
    :return:
    '''
    logging_file_dir = app.config['LOGGING_FILE_DIR']  # 日志文件所在的目录
    logging_file_max_bytes = app.config['LOGGING_FILE_MAX_BYTES']  # 日志文件最大的大小
    logging_file_backup = app.config['LOGGING_FILE_BACKUP']  # 保留备份的日志文件个数
    logging_level = app.config['LOGGING_LEVEL']  # 默认的日志级别

    # 设置日志的输出格式(针对文件）
    request_formatter = RequestShoppingFormatter(
        '[%(asctime)s] %(remote_addr)s  请求 %(url)s \t %(levelname)s 在 %(module)s %(lineno)d : %(message)s')

    # 检查如果目录不存在，则创建目录
    if os.path.isdir(logging_file_dir):
        pass
    else:
        os.mkdir(logging_file_dir)  # 如果目录不存在则创建

    # 自定义一个目录和日志文件。RotatingFileHandler : 按照指定的文件大小来规定日志文件的生产规则。
    '''这里这个handler是只有一个日志文件，所有产生的日志都是在financial.log这个文件里，也意味着，项目在启动以后，所有的进程在操作日志文件时，都操作的
    financial.log这个日志文件,会出现问题。解决办法：让一个进程操作一个文件。日志的文件名根据不同的进程名来命名，某一个进程在操作日志文件时只会操作对应的日志文件。
    参考另一个handler的日志文件命名规则就行了。
    backupCount是文件备份
    '''
    # flask_file_handler = logging.handlers.RotatingFileHandler(filename=os.path.join(logging_file_dir, 'financial.log'),
    #                                                           maxBytes=logging_file_max_bytes,
    #                                                           backupCount=logging_file_backup)

    # 为了让一个进程操作一个文件，所以文件名命名：加上当前进程的ID
    # TimedRotatingFileHandler:根据时间来规定日志文件的生成规则。
    flask_file_handler = logging.handlers.TimedRotatingFileHandler(
        filename=os.path.join(logging_file_dir, 'financial' + '_' + str(os.getpid()) + '.log'),
        when='D', interval=1, encoding='utf-8', backupCount=logging_file_backup
    )

    # 给当前的handler设置格式
    flask_file_handler.setFormatter(request_formatter)
    # 获取一个logger对象，根据包的名字
    flask_logger = logging.getLogger('financial')  # 这里只是得到了一个日志器对象，项目在运行的时候就会自动调用这个日志器进行日志的记录，我也不知道是为什么。
    # 将handler添加到logger中
    flask_logger.addHandler(flask_file_handler)
    # 设置logger的水平
    flask_logger.setLevel(logging_level)

    # 整个项目需要两个handler:文件，控制台
    # 创建一个控制控制台的handler
    flask_console_handler = logging.StreamHandler()
    # 给handler设置输出格式
    flask_console_handler.setFormatter(logging.Formatter('[%(asctime)s] %(levelname)s %(module)s %(lineno)d : %('
                                                         'message)s'))
    # 当前环境是debug模式时，才用控制台输出
    if app.debug:
        flask_logger.addHandler(flask_console_handler)
